/*
 * Copyright (c) 2019-2029, <a href="https://wwww.intellijboy.cn">https://wwww.intellijboy.cn</a> (liuburu@qq.com).
 * <p>
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 * <p>
 * <a href="https://www.apache.org/licenses/LICENSE-2.0">https://www.apache.org/licenses/LICENSE-2.0</a>
 * <p>
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */

package open.intellijboy.core.springboot;


import lombok.extern.slf4j.Slf4j;
import open.intellijboy.framework.core.exception.SystemException;
import org.springframework.boot.web.error.ErrorAttributeOptions;
import org.springframework.boot.web.servlet.error.DefaultErrorAttributes;
import org.springframework.web.context.request.WebRequest;

import java.util.HashMap;
import java.util.Map;

/**
 * 错误属性自定义
 *
 * @author intellijboy
 * @date 2023/10/2
 * @since 1.0
 */
@Slf4j
public class IntellijErrorAttributes extends DefaultErrorAttributes {

    @Override
    public Map<String, Object> getErrorAttributes(WebRequest webRequest, ErrorAttributeOptions options) {
        Map<String, Object> attributes = super.getErrorAttributes(webRequest, options);
        int status = Integer.parseInt(attributes.get("status")
                                                .toString());
        prettyLog(attributes);
        // 根据错误码进行增强
        Map<String, Object> exceptionMap = new HashMap<>();
        exceptionMap.put("code", status);
        Throwable error = this.getError(webRequest);
        if (error instanceof SystemException intellijEx) {
            exceptionMap.put("message", intellijEx.getErrorMessage());
        } else {
            exceptionMap.put("message", error.getMessage());
        }
        return exceptionMap;
    }

    /**
     * 日志打印
     *
     * @param attributes 错误属性
     */
    private void prettyLog(Map<String, Object> attributes) {
        Object timestamp = attributes.get("timestamp");
        log.error("**************************************{}********************************************************", timestamp);
        attributes.forEach((attr, obj) -> {
            log.error("{}:{}", attr, obj);
        });
        log.error("**************************************{}********************************************************", timestamp);
    }

}
